#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2017/8/17 0017 14:41
# @Author  : Swper
# @Site    : http://www.58jb.com/
# @File    : decorators.py
# @Software: PyCharm
from functools import wraps
from flask import session,redirect,url_for

#登陆限制的装饰器,如果没有登陆就不能跳转到管理页面
def login_required(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        if session.get('user_id'):
            return func(*args,**kwargs)
        else:
            return redirect(url_for('admin.login'))

    return wrapper